#####################################
##        PEinjector/utils         ##
#####################################

import os


# 找磁盘
def find_disk() -> str:  # This code from dashedgeless
    disk_list = "CDEFGHIJKABLMNOPQRSTUVWYZ"
    for i in disk_list:
        this_disk = i + ":"
        if os.path.exists(this_disk + "/PEinjector"):
            return this_disk
    raise OSError(
        "Cannot find the PEinjector disk!")
